package com.sky.aspect;

import com.sky.annotation.AutoFill;
import com.sky.context.BaseContext;
import com.sky.enumeration.OperationType;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import java.beans.PropertyDescriptor;
import java.lang.reflect.Field;
import java.time.LocalDateTime;

/**
 * 自定义切面，实现公共字段自动填充处理逻辑
 */
@Aspect
@Component
@Slf4j
public class AutoFillAspect {
    /**
     * 切入点
     */
//    @Pointcut("execution(* com.sky.mapper.*.*(..)) && @annotation(com.sky.annotation.AutoFill)")
//    public void autoFillPointCut() {}
    @Pointcut("execution(* com.sky.mapper.*.*(..)) && @annotation(com.sky.annotation.AutoFill)")
    public void pc() {}

    /**
     * 环绕通知
     */
    @Around("pc()&& @annotation(autoFill)")// &&@annotation(autoFill)将注解传入名为autoFill的形参
    public Object autoFill(ProceedingJoinPoint pjp,AutoFill autoFill) throws Throwable {
        //获取实体对象参数
        Object[] args = pjp.getArgs();
        Object arg = args[0];
        //获取操作类型
        OperationType value = autoFill.value();
        //判断其是做新增还是修改.
        Class c = arg.getClass();
        if (value == OperationType.INSERT) {
            //设置创建时间
//            Field createTime = c.getDeclaredField("createTime");
//            createTime.setAccessible(true);
//            createTime.set(arg,LocalDateTime.now());
            PropertyDescriptor pd = new PropertyDescriptor("createTime",c);//获取属性描述器
            pd.getWriteMethod().invoke(arg,LocalDateTime.now());//获取属性对应的setXxx方法,执行set方法.传参
            //设置创建人
            Field createUser = c.getDeclaredField("createUser");
            createUser.setAccessible(true);
            createUser.set(arg,BaseContext.getCurrentId());
        }
        //设置修改时间
        Field updateTime = c.getDeclaredField("updateTime");
        updateTime.setAccessible(true);
        updateTime.set(arg,LocalDateTime.now());
        //设置修改人
        Field updateUser = c.getDeclaredField("updateUser");
        updateUser.setAccessible(true);
        updateUser.set(arg,BaseContext.getCurrentId());

        Object proceed = pjp.proceed();
        return proceed;
    }



    /**
     * 前置通知,在通知中进行公共字段的赋值
     */
//    @Before("autoFillPointCut()")
//    public void autoFill(JoinPoint joinPoint) {
//        log.info("开始进行公共字段自动填充...");
//
//        //获取到当前被拦截的方法上的数据库操作类型
//        MethodSignature signature=(MethodSignature) joinPoint.getSignature();//方法签名对象
//        AutoFill autoFill = signature.getMethod().getAnnotation(AutoFill.class);//获得方法上的注解对象
//        OperationType operationType = autoFill.value();//获得数据库操作类型
//
//        //获取当前被拦截的方法的参数--实体对象
//        Object[] args = joinPoint.getArgs();
//        if (args == null || args.length == 0) {
//            return;
//        }
//
//        Object entity = args[0];
//
//        //准备赋值的数据
//        LocalDateTime now = LocalDateTime.now();
//        Long currentId = BaseContext.getCurrentId();
//
//        //根据当前不同的操作类型,为对应的属性通过反射来赋值
//        if (operationType == OperationType.INSERT) {
//            //为4个公共字段赋值
//            try {
//                Method setCreateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_TIME, LocalDateTime.class);
//                Method setCreateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_USER, Long.class);
//                Method setUpdateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_TIME, LocalDateTime.class);
//                Method setUpdateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_USER, Long.class);
//
//                //通过反射为对象属性赋值
//                setCreateTime.invoke(entity,now);
//                setCreateUser.invoke(entity,currentId);
//                setUpdateTime.invoke(entity,now);
//                setUpdateUser.invoke(entity,currentId);
//            } catch (Exception e) {
//                e.printStackTrace();
//            }
//        } else if (operationType == OperationType.UPDATE) {
//            //为2个公共字段赋值
//            try {
//                Method setUpdateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_TIME, LocalDateTime.class);
//                Method setUpdateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_USER, Long.class);
//
//                //通过反射为对象属性赋值
//                setUpdateTime.invoke(entity,now);
//                setUpdateUser.invoke(entity,currentId);
//            } catch (Exception e) {
//                e.printStackTrace();
//            }
//        }
//    }
}
